plots in python with matplotlip and numpy